# -*- coding: utf-8 -*-
"""
@author: einsam
"""
# python Attack_disturb.py

import os

# 测试代码（小数据集测试运行）
# os.system("python ./attacker.py \
#     --model_dir=../saved_models/best_acc_model \
#     --model_type=roberta \
#     --config_name=../base_models/microsoft/codebert-base\
#     --tokenizer_name=../base_models/microsoft/codebert-base \
#     --model_name_or_path=../base_models/microsoft/codebert-base \
#     --adv_store_path ../adv/Carrot_Bert.csv \
#     --disturb_store_path ./analysis/disturbed.csv \
#     --train_data=../preprocess/dataset/train.jsonl\
#     --valid_data=../preprocess/dataset/valid.jsonl\
#     --test_data=../preprocess/dataset/test_subs_0_10.jsonl \
#     --seed 123456  2>&1 | tee ../logs/Carrot_Bert.log")

# test_subs_*_*.jsonl 都来自Alert获得替换词时产生的子文件，但是不影响使用
# 这是我当前工作的一份子工作，而在我整体工作中，需要保持数据集一致，所以没有再另外编写分割test.jsonl的脚本
# 并且，Alert生成的替换词并没有在Carrot中起作用，正常使用即可
#

os.system("python ./attacker.py \
    --model_dir=../saved_models/best_acc_model \
    --model_type=roberta \
    --config_name=../base_models/microsoft/codebert-base\
    --tokenizer_name=../base_models/microsoft/codebert-base \
    --model_name_or_path=../base_models/microsoft/codebert-base \
    --adv_store_path ../adv/Carrot_Bert.csv \
    --disturb_store_path ./analysis/disturbed.csv \
    --train_data=../preprocess/dataset/train.jsonl\
    --valid_data=../preprocess/dataset/valid.jsonl\
    --test_data=../preprocess/dataset/test_subs_0_400.jsonl \
    --seed 123456  2>&1 | tee ../logs/Carrot_Bert.log")
